import torch

a = torch.FloatTensor([[0,1,2],[3,4,5]])
b = torch.FloatTensor([0,1,2])

# 1、矩阵乘法
print(a @ b)
print(a.matmul(b))

# 2、四则运算
print(a + b)
print(a - b)
print(a * b)
print(a / b)

# 3、求指数，根号，对数

# 4、逻辑运算

# 5、裁剪，限制数据上下界
print(a.clamp(2,4))